#include "workerManager.h"
#include "employee.h"
#include "manager.h"
#include "boss.h"

WorkerManager::WorkerManager()
{
    ifstream ifs;
    ifs.open(FILENAME, ios::in);

    // 文件不存在情况
    if (! ifs.is_open())
    {
        cout << "文件不存在" << endl;   // 测试输出
        this->m_EmpNum = 0;            // 初始化人数
        this->m_FileIsEmpty = true;    // 初始化文件为空标志
        this->m_EmpArray = NULL;       // 初始化数组
        ifs.close();                   // 关闭文件
        return ;
    }

    // 文件存在，并且没有记录
    char ch;
    ifs >> ch;
    if (ifs.eof())
    {
        cout << "文件为空" << endl;
        this->m_EmpNum = 0;
        this->m_FileIsEmpty = true;
        this->m_EmpArray = NULL;
        ifs.close();
        return;
    }

    // 文件存在，并且记录数据
    int num = this->get_EmpNum();
    cout << "职工人数为：" << num << endl;
    this->m_EmpNum = num;

    // 根据职工数创建数组
    this->m_EmpArray = new Worker *[this->m_EmpNum];
    // 初始化职工
    init_Emp();
    this->m_FileIsEmpty = false;
    // this->m_EmpArray = NULL;
    // ifs.close();
    // return;
}

WorkerManager::~WorkerManager()
{}

void WorkerManager::showMenu()
{
    cout << "****************************" << endl;
    cout << "*** 欢迎使用职工管理系统 ***" << endl;
    cout << "****** 0.退出关系程序 ******" << endl;
    cout << "****** 1.增加职工信息 ******" << endl;
    cout << "****** 2.显示职工信息 ******" << endl;
    cout << "****** 3.删除离职职工 ******" << endl;
    cout << "****** 4.修改职工信息 ******" << endl;
    cout << "****** 5.查找职工信息 ******" << endl;
    cout << "****** 6.按照编号排序 ******" << endl;
    cout << "****** 7.清空所有文档 ******" << endl;
    cout << "****************************" << endl;
}

void WorkerManager::exitSystem()
{
    cout << "欢迎下次使用" << endl;
    system("pause");
    exit(0);
}

void WorkerManager::AddEmp()
{
    cout << "请输入增加职工数量：" << endl;

    int addNum = 0;
    cin >> addNum;

    if (addNum > 0)
    {
        // 计算新空间大小
        int newSize = this->m_EmpNum + addNum;

        // 开辟新空间
        Worker ** newSpace = new Worker * [newSize];

        // 将源空间内容存放在新空间下
        if (this->m_EmpArray != NULL)
        {
            for (int i=0; i<this->m_EmpNum; i++)
            {
                newSpace[i] = this->m_EmpArray[i];
            }
        }

        // 输入新数据
        for (int i= 0; i<addNum; i++)
        {
            int id;
            string name;
            int dSelect;

            cout << "请输入第" << i+1 << "个新职工编号" << endl;
            cin >> id;

            cout << "请输入第" << i+1 << "个新职工姓名" << endl;
            cin >> name;

            cout << "请选择该职工的岗位：" << endl;
            cout << "1、普通职工" << endl;
            cout << "2、经理" << endl;
            cout << "3、老板" << endl;
            cin >> dSelect;

            Worker * worker = NULL;
            switch (dSelect)
            {
            case 1:
                worker = new Employee(id, name, dSelect);
                break;

            case 2:
                worker = new Manager(id, name, dSelect);
                break;

            case 3:
                worker = new Boss(id, name, dSelect);
                break;
            
            default:
                break;
            }

            newSpace[this->m_EmpNum + i] = worker;
        }

        // 释放原有空间
        delete[] this->m_EmpArray;

        // 更改新空间的指向
        this->m_EmpArray = newSpace;

        // 更新新的个数
        this->m_EmpNum = newSize;

        // 提示信息
        cout << "成功添加" << addNum << "名新员工" << endl;

        // 保存到文件
        this->save();
    }
    else
    {
        cout << "输入有误" << endl;
    }
    
    system("pause");
    system("cls");
}

void WorkerManager::save()
{
    ofstream ofs;
    ofs.open(FILENAME, ios::out);

    for (int i=0; i<this->m_EmpNum; i++)
    {
        ofs << this->m_EmpArray[i]->m_Id << "  ";
        ofs << this->m_EmpArray[i]->m_Name << "  ";
        ofs << this->m_EmpArray[i]->m_DeptId << endl;
    }

    ofs.close();
}


int WorkerManager::get_EmpNum()
{
    ifstream ifs;
    ifs.open(FILENAME, ios::in);

    int id;
    string name;
    int dId;

    int num = 0;
    while (ifs >> id && ifs >> name && ifs >>dId)
    {
        // 记录人数
        num++;
    }

    ifs.close();

    return num;
}

void WorkerManager::init_Emp()
{
    ifstream ifs;
    ifs.open(FILENAME, ios::in);

    int id;
    string name;
    int dId;

    int index = 0;
    while (ifs>>id && ifs>>name && ifs>>dId)
    {
        Worker * worker = NULL;

        if (dId == 1)
        {
            worker = new Employee(id, name, dId);
        }
        else if (dId == 2)
        {
            worker = new Manager(id, name, dId);
        }
        else
        {
            worker = new Boss(id, name, dId);
        }

        this->m_EmpArray[index] = worker;
        index ++;      
    }
    ifs.close();
}

void WorkerManager::show_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空！" << endl;
    }
    else
    {
        for(int i=0; i<m_EmpNum; i++)
        {
            this->m_EmpArray[i]->showInfo();
        }
    }

    system("pause");
    system("cls");
    
}

int WorkerManager::isExist(int id)
{
    int index = -1;

    for (int i=1; i<this->m_EmpNum; i++)
    {
        if (this->m_EmpArray[i]->m_Id == id)
        {
            index = i;
            break;
        }
    }

    return index;
}

void WorkerManager::del_Emp()
{
    if(this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空" << endl;
    }
    else
    {
        cout << "请输入想要删除的职工编号：" << endl;
        int id = 0;
        cin >> id;

        int index = this->isExist(id);

        if (index != -1)
        {
            for (int i=index; i<this->m_EmpNum; i++)
            {
                this->m_EmpArray[i] = this->m_EmpArray[i+1];
            }
            this->m_EmpNum --;
            this->save();
            cout << "删除成功" << endl;
        }
        else
        {
            cout << "删除失败，未找到该员工" << endl;
        }
    }
    system("pause");
    system("cls");
}

void WorkerManager::mod_Emp()
{
    if(this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空" << endl;
    }
    else
    {
        cout << "请输入待修改职工编号" << endl;
        int id ;
        cin >> id;

        int ret = this->isExist(id);
        if(ret != -1)
        {
            delete this->m_EmpArray[ret];

            int newId = 0;
            string newName = "";
            int dSelect = 0;

            cout << "查到：" << id << "号职工，请输入新职工号" << endl;
            cin >> newId;

            cout << "请输入新姓名：" << endl;
            cin >> newName;

            cout << "请输入岗位：" << endl;
            cout << "1.普通职工" << endl;
            cout << "2.经理" << endl;
            cout << "3.老板" << endl;
            cin >> dSelect;

            Worker * worker = NULL;
            switch (dSelect)
            {
            case 1:
                worker = new Employee(newId, newName, dSelect);
                break;

            case 2:
                worker = new Manager(newId, newName, dSelect);
                break;

            case 3:
                worker = new Boss(newId, newName, dSelect);
                break;
            
            default:
                break;
            }

            // 更改数据到数组
            this->m_EmpArray[ret] = worker;

            cout << "修改成功" << this->m_EmpArray[ret]->m_DeptId << endl;

            // 保存文件
            this->save();
        }
        else
        {
            cout << "修改事变，查无此人" << endl;
        }
    }
    system("pause");
    system("cls");
}

void WorkerManager::find_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空" << endl;
    }
    else
    {
        cout << "请输入查找的方式： \t 1.按职工编号查找 \t 2.按职工姓名查找" << endl;
        int select = 0;
        cin >> select;

        if (select == 1)
        {
            int id;
            cout << "请输入查找的职工编号：" << endl;
            cin >> id;

            int ret = isExist(id);
            if (ret != -1)
            {
                cout << "查找成功" << endl;
                this->m_EmpArray[ret]->showInfo();
            }
            else
            {
                cout << "查找失败，查无此人" << endl;
            }
        }
        else if (select == 2)
        {
            string name;
            cout << "请输入查找的姓名" << endl;
            cin >> name;

            bool flag = false;
            for (int i=0; i<m_EmpNum; i++)
            {
                if(m_EmpArray[i]->m_Name == name)
                {
                    cout << "查找成功，职工编号为：" << m_EmpArray[i]->m_Id << "号的信息如下：" << endl;

                    flag = true;
                    this->m_EmpArray[i]->showInfo();
                }
            }

            if (flag == false)
            {
                cout << "查找失败，查无此人" << endl;
            }
        }
        else
        {
            cout << "输入选项有误" << endl;
        }
    }
    system("pause");
    system("cls");
}

void WorkerManager::sort_Emp()
{
    if (this->m_FileIsEmpty)
    {
        cout << "文件不存在或记录为空" << endl;
        system("pause");
        system("cls");
    }
    else
    {
        cout << "请选择排序方式：\t 1.按职工号升序 \t 2.按职工号降序" << endl;
        int select = 0;
        cin >> select;

        for (int i=0; i<this->m_EmpNum; i++)
        {
            int minOrMax = i;
            for (int j=i+1; j<m_EmpNum; j++)
            {
                if(select = 1)
                {
                    if(m_EmpArray[minOrMax]->m_Id > m_EmpArray[j]->m_Id)
                    {
                        minOrMax = j;
                    }
                }
                else
                {
                    if(m_EmpArray[minOrMax]->m_Id < m_EmpArray[j]->m_Id)
                    {
                        minOrMax = j;
                    }
                } 
            }
            if (i != minOrMax)
            {
                Worker * temp = m_EmpArray[i];
                m_EmpArray[i] = m_EmpArray[minOrMax];
                m_EmpArray[minOrMax] = temp;
            }
        }
        cout << "排序成功" << endl;
        this->save();
        this->show_Emp();
    }
    
}

void WorkerManager::clear_File()
{
    cout << "确认清空吗？" << endl;
    cout << "1.确定" << endl;
    cout << "2.返回" << endl;

    int select = 0;
    cin >> select;

    if (select == 1)
    {
        ofstream ofs(FILENAME, ios::trunc);
        ofs.close();

        if (this->m_EmpArray != NULL)
        {
            for (int i=0; i<m_EmpNum; i++)
            {
                if (this->m_EmpArray[i] != NULL)
                {
                    delete this->m_EmpArray[i];
                }
            }
            this->m_EmpNum = 0;
            delete[] this->m_EmpArray;
            this->m_EmpArray = NULL;
            this->m_FileIsEmpty = true;
        }
        cout << "清空成功" << endl;
    }
    system("pause");
    system("cls");
}